Hyper parameter tuning of neural network in python